const { sequelize } = require('../config/database');

async function updateServerTypeEnum() {
  try {
    console.log('开始更新token_transactions表的server_type字段ENUM值...');
    
    // 使用原生SQL来修改ENUM字段
    await sequelize.query(`
      ALTER TABLE token_transactions 
      MODIFY COLUMN server_type 
      ENUM('text', 'img', 'voice', 'video', 'subscription', 'disposable','promotional') 
      COMMENT '服务类型：文本生成、图片生成、声音生成、视频生成、订阅、一次性购买、活动赠送'
    `);
    
    console.log('server_type字段ENUM值更新成功！');
    console.log('新增类型: subscription, disposable');
    
  } catch (error) {
    console.error('更新server_type字段失败:', error);
    throw error;
  }
}

// 如果直接运行此脚本
if (require.main === module) {
  updateServerTypeEnum()
    .then(() => {
      console.log('迁移完成');
      process.exit(0);
    })
    .catch((error) => {
      console.error('脚本执行失败:', error);
      process.exit(1);
    });
}

module.exports = { updateServerTypeEnum }; 